#!/bin/bash
set -e

export LOG_FILE="info.csv"

# LOG_AFTER_STEP_NUM="20" python3 enflame_model_test.py --is_training=True \
#     --device=gpu --model=resnet --dtype=bf16 \
#     --data_format=NHWC  --dataset=imagenet --batch_size=256 \
#     --epoch=90 --enable_saver=True --enable_evaluate=True \
#     --optimizer=sgd

# LOG_AFTER_STEP_NUM="20" python3 enflame_model_test.py --is_training=True \
#     --device=gpu --model=resnet --dtype=fp16 \
#     --data_format=NHWC  --dataset=imagenet --batch_size=256 \
#     --epoch=90 --enable_saver=True --enable_evaluate=True \
#     --optimizer=sgd


LOG_AFTER_STEP_NUM="100" python3 enflame_model_test.py --is_training=True \
    --device=gpu --model=resnet --dtype=fp32 \
    --data_format=NHWC  --dataset=imagenet --batch_size=64 \
    --epoch=90 --enable_saver=True --enable_evaluate=True \
    --optimizer=sgd

# LOG_AFTER_STEP_NUM="20" python3 enflame_model_test.py --is_training=True \
#     --device=gpu --model=inception_v3 --dtype=bf16 \
#     --data_format=NHWC  --dataset=imagenet --batch_size=128 \
#     --epoch=90 --enable_saver=True --enable_evaluate=True \
#     --optimizer=sgd

# LOG_AFTER_STEP_NUM="20" python3 enflame_model_test.py --is_training=True \
#     --device=gpu --model=inception_v3 --dtype=fp16 \
#     --data_format=NHWC  --dataset=imagenet --batch_size=128 \
#     --epoch=90 --enable_saver=True --enable_evaluate=True \
#     --optimizer=sgd

LOG_AFTER_STEP_NUM="100" python3 enflame_model_test.py --is_training=True \
    --device=gpu --model=inception_v3 --dtype=fp32 \
    --data_format=NHWC  --dataset=imagenet --batch_size=64 \
    --epoch=90 --enable_saver=True --enable_evaluate=True \
    --optimizer=sgd
